# registry/model_registry.py

from typing import Dict, Type

from app.services.doubao.interface import UBALDOInterface

MODEL_REGISTRY: Dict[str, Type[UBALDOInterface]] = {}


def register_model(name: str):
    def wrapper(cls):
        MODEL_REGISTRY[name] = cls
        return cls

    return wrapper
